python=3.6.9
numpy=1.17.4
torch=1.3.1
CUDA=10.1
